'''
此模块用于处理指定目录中的Excel文件路径处理相关的问题。

1.获取指定目录下的所有文件。
2.岷县18个乡镇名称枚举列表。
'''
import os
import os.path
import pandas as pd

min_town_list = ['西寨','十里','清水','岷阳','秦许','茶埠','禾驮','寺沟','西江','梅川','维新','中寨','申都','蒲麻','闾井','锁龙','马坞','麻子川']

def get_all_excel_files(path):
    '''获取指定目录下的所有文件
    遍历指定目录下的所有xls及xlsx文件，并以DataFrame的格式返回。
    '''
    file_df = pd.DataFrame(columns=['文件名称','文件路径'])
    for root, dirs, files in os.walk(path):
        for file in files:
            if os.path.splitext(file)[1] in ['.xls','.xlsx']:
                file_path = os.path.join(root,file)
                file_df.loc[file_df.shape[0],:] = os.path.splitext(os.path.basename(file))[0],file_path
    return file_df